package config

type LLMConfig struct {
	Model       string    `json:"model"`
	ApiUrl      string    `json:"ApiUrl"`
	ApiKey      string    `json:"apiKey"`
	Mssages     []Message `json:"messages"`
	Temperature float32   `json:"temperature"`
	Stream      bool      `json:"stream"`
}

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

// deepseek-chat
func NewLLMConfig() *LLMConfig {
	return &LLMConfig{
		Model:       "deepseek-chat",
		ApiUrl:      "https://api.deepseek.com/v1/chat/completions",
		ApiKey:      "sk-9c5e1f57430143048b9a1cce891e91b9",
		Temperature: 0.5,
		Stream:      true,
	}
}

// 豆包
func NewDouBaoLLMConfig() *LLMConfig {
	return &LLMConfig{
		Model:       "ep-20240828180738-nkv56",
		ApiKey:      "b340af9e-ecab-4dfb-8abe-8997907ec3de",
		Temperature: 0.5,
		Stream:      true,
	}
}
